package com.marmot.cloud.framework.core.tools;

import lombok.AccessLevel;
import lombok.NoArgsConstructor;
import org.springframework.beans.BeansException;
import org.springframework.context.ApplicationContext;
import org.springframework.context.ApplicationContextAware;
import org.springframework.stereotype.Component;

/**
 * 应用程序上下文工具类，用于获取 Spring 应用程序上下文和管理 Bean 对象。
 * 通过实现 ApplicationContextAware 接口，可以在 Spring 容器启动时自动注入 ApplicationContext。
 */
@Component
@NoArgsConstructor(access = AccessLevel.PRIVATE)
public class AppCtxUtils implements ApplicationContextAware {

    // 静态变量保存 ApplicationContext
    private static ApplicationContext applicationContext;

    /**
     * 实现 ApplicationContextAware 接口的方法，注入 ApplicationContext。
     *
     * @param applicationContext Spring 应用程序上下文
     * @throws BeansException 如果注入 ApplicationContext 出现异常，抛出 BeansException 异常
     */
    @Override
    public void setApplicationContext(ApplicationContext applicationContext) throws BeansException {
        AppCtxUtils.applicationContext = applicationContext;
    }

    /**
     * 获取应用程序上下文。
     *
     * @return ApplicationContext 对象
     */
    public static ApplicationContext getApplicationContext() {
        return applicationContext;
    }

    /**
     * 根据 bean 名称获取对应的 bean 对象。
     *
     * @param name bean 的名称
     * @return 对应名称的 bean 对象
     * @throws BeansException 如果获取 bean 对象出现异常，抛出 BeansException 异常
     */
    public static <T> T getBean(String name) throws BeansException {
        return (T) applicationContext.getBean(name);
    }

    /**
     * 根据 bean 的类型获取对应的 bean 对象。
     *
     * @param clazz bean 的类型
     * @return 对应类型的 bean 对象
     * @throws BeansException 如果获取 bean 对象出现异常，抛出 BeansException 异常
     */
    public static <T> T getBean(Class<T> clazz) throws BeansException {
        return applicationContext.getBean(clazz);
    }

    /**
     * 判断容器中是否包含指定名称的 bean。
     *
     * @param name bean 的名称
     * @return 如果容器中包含指定名称的 bean，则返回 true；否则返回 false
     */
    public static boolean containsBean(String name) {
        return applicationContext.containsBean(name);
    }
}
